#! /usr/bin/env python
# -*- encoding: utf-8 -*-
'''
Test suite on word_count
'''
import unittest
from word_count import WordCount

class WcTestCase(unittest.TestCase):
    
    def setUp(self):
        self.wc = WordCount()
 
    def tearDown(self):
        self.wc = None
  
    def test_get_line_count(self):
        self.assertEqual(self.wc.get_line_count("test"),6,'test line count fail')

    def test_get_char_count(self):
        self.assertEqual(self.wc.get_char_count("test"),30,'test char count fail')
  
    def test_get_word_count(self):
        self.assertEqual(self.wc.get_word_count("test"),8,'test word count fail')

if __name__=='__main__':

    ''' First method of using unittest'''
    #suite = unittest.TestSuite()
    #suite.addTest(WcTestCase("test_get_line_count"))
    #suite.addTest(WcTestCase("test_get_char_count"))
    #suite.addTest(WcTestCase("test_get_word_count"))   

    #runner = unittest.TextTestRunner()
    #runner.run(suite)

    ''' Second method of using unittest'''
    unittest.main()
